package org.dromara.web.listener;

import cn.dev33.satoken.config.SaTokenConfig;
import cn.dev33.satoken.listener.SaTokenListener;
import cn.dev33.satoken.stp.SaLoginModel;
import cn.hutool.http.useragent.UserAgent;
import cn.hutool.http.useragent.UserAgentUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.dromara.common.core.constant.CacheConstants;
import org.dromara.common.core.constant.Constants;
import org.dromara.common.core.domain.dto.UserOnlineDTO;
import org.dromara.common.core.utils.MessageUtils;
import org.dromara.common.core.utils.ServletUtils;
import org.dromara.common.core.utils.SpringUtils;
import org.dromara.common.core.utils.ip.AddressUtils;
import org.dromara.common.log.event.LogininforEvent;
import org.dromara.common.redis.utils.RedisUtils;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.tenant.helper.TenantHelper;
import org.dromara.web.service.SysLoginService;
import org.springframework.stereotype.Component;

import java.time.Duration;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：用户行为 侦听器的实现
 */
/**
 * 用户行为监听器类，用于实现SaTokenListener接口，监听用户的相关操作
 * 此类主要用于捕捉和处理用户登录、注销等行为，以便进行相应的业务逻辑处理
 */
@RequiredArgsConstructor
@Component
@Slf4j
public class UserActionListener implements SaTokenListener {

    /**
     * Sa-Token 配置类 Model
     */
    private final SaTokenConfig tokenConfig;

    /**
     * 登录校验方法
     */
    private final SysLoginService loginService;

    /**
     * 每次登录时触发
     * 执行登录操作，根据登录类型和提供的登录信息完成用户登录流程
     *
     * @param loginType 登录类型，例如用户名、手机号等
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     * @param loginModel 登录模型，包含登录额外信息和配置
     */
    @Override
    public void doLogin(String loginType, Object loginId, String tokenValue, SaLoginModel loginModel) {
        // 解析用户代理信息，用于后续记录浏览器和操作系统信息
        UserAgent userAgent = UserAgentUtil.parse(ServletUtils.getRequest().getHeader("User-Agent"));
        String ip = ServletUtils.getClientIP();
        UserOnlineDTO dto = new UserOnlineDTO();
        dto.setIpaddr(ip);
        dto.setLoginLocation(AddressUtils.getRealAddressByIP(ip));
        dto.setBrowser(userAgent.getBrowser().getName());
        dto.setOs(userAgent.getOs().getName());
        dto.setLoginTime(System.currentTimeMillis());
        dto.setTokenId(tokenValue);
        String username = (String) loginModel.getExtra(LoginHelper.USER_NAME_KEY);
        String tenantId = (String) loginModel.getExtra(LoginHelper.TENANT_KEY);
        dto.setUserName(username);
        dto.setClientKey((String) loginModel.getExtra(LoginHelper.CLIENT_KEY));
        dto.setDeviceType(loginModel.getDevice());
        dto.setDeptName((String) loginModel.getExtra(LoginHelper.DEPT_NAME_KEY));
        // 动态设置租户信息，并存储登录用户信息到Redis
        TenantHelper.dynamic(tenantId, () -> {
            if(tokenConfig.getTimeout() == -1) {
                RedisUtils.setCacheObject(CacheConstants.ONLINE_TOKEN_KEY + tokenValue, dto);
            } else {
                RedisUtils.setCacheObject(CacheConstants.ONLINE_TOKEN_KEY + tokenValue, dto, Duration.ofSeconds(tokenConfig.getTimeout()));
            }
        });
        // 记录登录日志
        LogininforEvent logininforEvent = new LogininforEvent();
        logininforEvent.setTenantId(tenantId);
        logininforEvent.setUsername(username);
        logininforEvent.setStatus(Constants.LOGIN_SUCCESS);
        logininforEvent.setMessage(MessageUtils.message("user.login.success"));
        logininforEvent.setRequest(ServletUtils.getRequest());
        SpringUtils.context().publishEvent(logininforEvent);
        // 更新登录信息
        loginService.recordLoginInfo((Long) loginModel.getExtra(LoginHelper.USER_KEY), ip);
        log.info("user doLogin, userId:{}, token:{}", loginId, tokenValue);
    }

    /**
     * 每次注销时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     */
    @Override
    public void doLogout(String loginType, Object loginId, String tokenValue) {
        RedisUtils.deleteObject(CacheConstants.ONLINE_TOKEN_KEY + tokenValue);
        log.info("user doLogout, userId:{}, token:{}", loginId, tokenValue);
    }

    /**
     * 每次被踢下线时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     */
    @Override
    public void doKickout(String loginType, Object loginId, String tokenValue) {
        RedisUtils.deleteObject(CacheConstants.ONLINE_TOKEN_KEY + tokenValue);
        log.info("user doKickout, userId:{}, token:{}", loginId, tokenValue);
    }

    /**
     * 每次被顶下线时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     */
    @Override
    public void doReplaced(String loginType, Object loginId, String tokenValue) {
        RedisUtils.deleteObject(CacheConstants.ONLINE_TOKEN_KEY + tokenValue);
        log.info("user doReplaced, userId:{}, token:{}", loginId, tokenValue);
    }

    /**
     * 每次被封禁时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param level 禁用级别，表示禁用的严格程度，高级别可能包括低级别禁用的所有功能并增加额外限制
     * @param disableTime 禁用时间，表示该登录方式被禁用的时长，单位为秒
     */
    @Override
    public void doDisable(String loginType, Object loginId, String service, int level, long disableTime) {
    }

    /**
     * 每次被解封时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param service 服务名称，表示该登录方式所属的服务或应用
     */
    @Override
    public void doUntieDisable(String loginType, Object loginId, String service) {
    }

    /**
     * 每次打开二级认证时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     * @param service 服务名称，表示该登录方式所属的服务或应用
     * @param safeTime 安全时间，表示操作被认为安全之前需要等待的时间
     */
    @Override
    public void doOpenSafe(String loginType, String tokenValue, String service, long safeTime) {
    }

    /**
     * 每次创建Session时触发
     * @param loginType 登录类型，例如用户名、手机号等
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     * @param service 服务名称，表示该登录方式所属的服务或应用
     */
    @Override
    public void doCloseSafe(String loginType, String tokenValue, String service) {
    }

    /**
     * 每次创建Session时触发
     * @param id id
     */
    @Override
    public void doCreateSession(String id) {
    }

    /**
     * 每次注销Session时触发
     * @param id id
     */
    @Override
    public void doLogoutSession(String id) {
    }

    /**
     * 每次Token续期时触发
     * @param tokenValue 令牌值，用于唯一标识此次登录会话
     * @param loginId 登录标识，具体类型取决于登录类型
     * @param timeout 新的超时时间，单位为毫秒，用于设置会话的超时时间
     */
    @Override
    public void doRenewTimeout(String tokenValue, Object loginId, long timeout) {
    }
}
